#!/usr/bin/python

from hips.scheduling import schedule_mocks


class MockDomain(object):

  def __init__(self):
    self.expected = []
    self.actual = []

  def ExpectHideCall(self, args):
    self.expected.append(args)

  def hideValue(self, value):
    self.actual.append(value)

  def Verify(self, testcase):
    testcase.assertListEqual(self.expected, self.actual)


def CreateMockConstraintParameters(assignment):
  variables = range(len(assignment))
  domains = MockDomain()
  mock_assignments = {}
  for v in variables:
    mock_assignments[v] = assignment[v]
  return variables, domains, mock_assignments
